coding a atm python